import os
import sys

from numpy import double

from data_manage import check_data, check_digits, check_central_meridian, check_east_north


def write_data(data):
    # 将数据写入excel文件，如果文件不存在则创建一个新文件，已存在则覆盖
    # 指定excel文件路径
    excel_path = '临时坐标.xlsx'
    # 判断文件是否存在
    if os.path.isfile(excel_path):
        # 文件存在，删除原文件
        os.remove(excel_path)
    # 将数据写入CSV文件
    data.to_excel(excel_path, index=False)


if __name__ == '__main__':
    src_file_path = sys.argv[1]  # 源文件路径
    src_coorSys_text = sys.argv[3]  # 源坐标系文字
    src_central_meridian = double(sys.argv[4])  # 源中央子午线
    src_a = double(sys.argv[5])  # 源长半轴
    src_rf = double(sys.argv[6])  # 源扁率

    data = check_data(src_file_path)
    # 检查数据位数是否正确
    check_digits(data)
    # 检查东北、北东格式
    check_east_north(data)
    # 检查中央子午线和带号是否匹配
    check_central_meridian(data, src_coorSys_text, src_central_meridian)
    # 写入临时数据供使用
    write_data(data)
